package com.fhyc.dto;


import lombok.Data;

import java.util.ArrayList;
import java.util.List;

/**
 * 母线、厂站、控制区树形结构
 */
@Data
public class BusBarNaviDto {
    /**
     * ID
     */
    private String id;
    /**
     * 名称
     */
    private String name;
    /**
     * 节点类型：1-控制区 2-厂站 3-母线 4-小电源
     */
    private Integer type;
    /**
     * 排序 type=1的排在最前面 type=2的排在第二 type=3的排在第三 type=4的排在最后
     */
    private Integer sort;
    /**
     * 子节点
     */
    private List<BusBarNaviDto> children;

    public BusBarNaviDto(String id, String name, int type) {
        this.id = id;
        this.name = name;
        this.type = type;
        this.sort = -1;
        this.children = new ArrayList<>();
    }

    public void addChild(BusBarNaviDto child) {
        this.children.add(child);
    }
}
